Tensorflow GPU版本安装

第一步,下载安装CUDA

1 Tensorflow 1.8.0,对应CUDA9.0版。

2 安装前如果C:\Program Files下有NVIDIA Corporation等文件夹,先删除。

3 添加环境变量:

1
2
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64

4 注销后,重新登录让环境变量生效。

第二步,下载安装cudnn

1 这个相当于CUDA补丁,下载需要注册NVIDIA账号。

2 下载完后解压,拷贝到CUDA安装目录。

第三步,安装Tensorflow GPU版本

1 先卸载旧版,防止冲突。

2 然后,执行下面命令,下载最新版1.8.0 的Tensorflow

1
pip install tensorflow-gpu

꧁༺The༒End༻꧂